﻿#region License
/***********************************************
 *  WCFReactiveX v.1
 *  Written by James Levingston
 *  
 *  To get latest version or to see latest info:
 *  http://wcfreactivex.codeplex.com/
 *  
 *  The MIT License (MIT)
 *  http://en.wikipedia.org/wiki/MIT_License
 *  Copyright (C) <2011> by <James Levingston>
 *
 *  Permission is hereby granted, free of charge, to any person obtaining a copy
 *  of this software and associated documentation files (the "Software"), to deal
 *  in the Software without restriction, including without limitation the rights
 *  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 *  copies of the Software, and to permit persons to whom the Software is
 *  furnished to do so, subject to the following conditions:
 *
 *  The above copyright notice and this permission notice shall be included in
 *  all copies or substantial portions of the Software.

 *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 *  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 *  THE SOFTWARE.
 ***/
#endregion

using System;
using System.Configuration;
using System.Diagnostics;
using System.ServiceModel;
using System.ServiceModel.Dispatcher;
using System.ServiceModel.Configuration;
using System.ServiceModel.Description;

namespace WCFReactiveX
{
    internal class ProviderMessageInspector : IClientMessageInspector
    {
        private Stopwatch watch = null;
        private string currentRequestAction = null;

        private IProviderEventListener adapter = null;

        public ProviderMessageInspector(IProviderEventListener adapter) { this.adapter = adapter; }

        #region IClientMessageInspector Members

        public void AfterReceiveReply(ref System.ServiceModel.Channels.Message reply, object correlationState)
        {
            this.LogResponse(reply);
        }

        public object BeforeSendRequest(ref System.ServiceModel.Channels.Message request, IClientChannel channel)
        {
            this.LogRequest(request);
            return null;
        }

        #endregion

        #region WCF Metrics code
        private void LogRequest(System.ServiceModel.Channels.Message request)
        {
            var timerStart = string.Empty;
            if (this.IsMetricsActivated())
            {
                this.currentRequestAction = request.Headers.Action;
                this.watch = new Stopwatch();
                this.watch.Start();
                timerStart = DateTime.Now.ToString() + ":";
            }

            this.adapter.Request(timerStart + request.ToString());
        }
        private void LogResponse(System.ServiceModel.Channels.Message response)
        {
            var timerEnd = string.Empty;
            if (this.IsMetricsActivated())
            {
                this.watch.Stop();
                timerEnd = DateTime.Now.ToString() + ": Total Call Time (MS) : " + this.watch.ElapsedMilliseconds;
            }

            this.adapter.Response(timerEnd + response.ToString());
        }

        private bool IsMetricsActivated()
        {
            var metrics = this.GetConfigEntry("WFCMetricsActivated");
            if (metrics == null)
                return false;

            if (metrics.ToLower() == "true")
                return true;

            return false;
        }
        #endregion
    }
}
